import json
import random


class 双色球开奖():

    def __init__(self):
        self.红色号码 = []
        self.蓝色号码 = []
        for i in range(1, 34):
            self.红色号码.append(i)
        for i in range(1, 17):
            self.蓝色号码.append(i)

    @staticmethod
    def 洗牌(list, count=1):
        # 打乱列表顺序
        while count > 0:
            count -= 1
            random.shuffle(list)

    def 生成双色球号码(self, count=1):
        红色号码列表 = self.红色号码
        蓝色号码列表 = self.蓝色号码
        result_list = []
        while count > 0:
            count -= 1
            # 洗牌50次
            self.洗牌(红色号码列表, 50)
            self.洗牌(蓝色号码列表, 50)
            # 摇奖
            红色中奖号码 = random.sample(红色号码列表, 7)
            蓝色中奖号码 = random.sample(蓝色号码列表, 2)
            红色中奖号码.sort()
            蓝色中奖号码.sort()
            print(f'红色：{红色中奖号码} 蓝色：{蓝色中奖号码}')
            dict1 = {
                "hong": 红色中奖号码,
                "lan": 蓝色中奖号码
            }
            # result_list.append(json.dumps(dict1))  #flask需要将字典转换为JSON传给前端
            result_list.append(dict1)  # Django直接传字典给前端，前端通过模板语法接收即可
        return result_list


if __name__ == '__main__':
    中1000W = 双色球开奖()
    x = int(input("请输入生成号码注数："))
    中1000W.生成双色球号码(x)
